package com.uyayo.framework.web.resolver;


import com.uyayo.framework.web.annotation.GetRequestInfo;
import com.uyayo.framework.web.support.RequestBaseHelper;
import org.springframework.core.MethodParameter;
import org.springframework.web.bind.support.WebDataBinderFactory;
import org.springframework.web.context.request.NativeWebRequest;
import org.springframework.web.context.request.ServletWebRequest;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.method.support.ModelAndViewContainer;

import javax.servlet.http.HttpServletRequest;


/**
 * 获取用户请求的requestInfo信息
 * @author Best
 */
public class RequestInfoDefaultArgumentResolver implements HandlerMethodArgumentResolver {

    protected RequestBaseHelper requestBaseHelper;

    public RequestBaseHelper getRequestBaseHelper() {
        return requestBaseHelper;
    }

    public void setRequestBaseHelper(RequestBaseHelper requestBaseHelper) {
        this.requestBaseHelper = requestBaseHelper;
    }

    public RequestInfoDefaultArgumentResolver(RequestBaseHelper requestBaseHelper) {
        this.requestBaseHelper = requestBaseHelper;
    }

    @Override
    public boolean supportsParameter(MethodParameter parameter) {
        return parameter.hasParameterAnnotation(GetRequestInfo.class);

    }

    @Override
    public Object resolveArgument(MethodParameter parameter,
                                  ModelAndViewContainer mavContainer,
                                  NativeWebRequest webRequest,
                                  WebDataBinderFactory binderFactory) {
        HttpServletRequest request = ((ServletWebRequest) webRequest).getRequest();
        return requestBaseHelper.getRequestInfo(request);

    }
}
